#ifndef MNN_INFER_API_H
#define MNN_INFER_API_H

#include <iostream>
#include <string>
#include <opencv2/opencv.hpp>
#include <MNN/Interpreter.hpp>
#include "MNN/ImageProcess.hpp"
#include "MNN/MNNDefine.h"
#include "MNN/Tensor.hpp"

class mnnModel
{
public:

	int output_size=1;
	float *prob=nullptr;
	int OUTPUT_CANDIDATES;

    mnnModel();
    ~mnnModel();
    
    void Init(std::string model_path,std::string input_blob_name,std::string out_blob_name);
    void doInference(float *input_arr);

private:
	std::shared_ptr<MNN::Interpreter> mnnNet;
	MNN::Session* session{nullptr};
    MNN::Tensor* input{nullptr};
	MNN::Tensor* output{nullptr};
   
};



#endif
